package com.wxg.ch06;

public class BSTv2<E extends Comparable<E>> {

    private Node root;
    private int size;

    public BSTv2() {
        root = null;
        size = 0;
    }

    public int size() {
        return size;
    }

    public boolean isEmpty() {
        return size == 0;
    }

    public void add(E e) {
        root = add(root, e);
    }

    /**
     * 这是一段很高明的代码逻辑
     *
     * 不会存在 left 或 right 有值而被赋予新值的情况(除非相等)： 因为return的条件是节点为`null`
     *      - 也就是说只要节点有值，就会向下递归，直到找到空节点，然后赋值并退出递归。
     *      - 这样根节点为空的情形也不用判断了。 比如，`root = add(root, e)` 不用判断 root 是否为空
     * 节点和当前节点相等时，会被重新赋值一次，这是例外。
     *
     * @param node
     * @param e
     * @return
     */
    private Node add(Node node, E e) {
        if (node == null) {
            size++;
            return new Node(e);
        }

        if (node.e.compareTo(e) < 0) {
            node.left = add(node.left, e);
        } else if (node.e.compareTo(e) > 0) {
            node.right = add(node.right, e);
        }

        // node.e.compareTo(e) == 0 什么也不做

        return node;
    }

    private class Node {
        public E e;
        public Node left, right;

        public Node(E e) {
            this.e = e;
            left = null;
            right = null;
        }
    }
}
